package leo.mystudy.leetcode;

/**
 * @author lichao email:lichao@witspring.com
 * @Description: 2. 两数相加
 * @Date 2018/4/25 19:14
 * @since Ver V0.0.1
 */
public class AddTwoNumbers {
    //    给定两个非空链表来表示两个非负整数。位数按照逆序方式存储，它们的每个节点只存储单个数字。将两数相加返回一个新的链表。
//
//    你可以假设除了数字 0 之外，这两个数字都不会以零开头。
//
//    示例：
//
//    输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
//    输出：7 -> 0 -> 8
//    原因：342 + 465 = 807

    public static void main(String[] args) {
        new AddTwoNumbers().test();
    }

    public void test() {
        ListNode l1 = new ListNode(2);
        l1.next = new ListNode(4);
        l1.next.next = new ListNode(3);
        ListNode l2 = new ListNode(5);
        l2.next = new ListNode(6);
        l2.next.next = new ListNode(4);
        ListNode result = addTwoNumbers(l1, l2);
        printListNode(result);
    }

    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode result = new ListNode(0);
        ListNode tmpNode = result;
        int a;
        int b;
        while (l1 != null || l2 != null) {
            {
                // 如果l1不为空，a=l1的值，l1指向next，否则a=0。
                if (l1 != null) {
                    a = l1.val;
                    l1 = l1.next;
                } else {
                    a = 0;
                }
                // 如果l2不为空，b=l2的值，l2指向next，否则b=0。
                if (l2 != null) {
                    b = l2.val;
                    l2 = l2.next;
                } else {
                    b = 0;
                }
                // 求和
                int temp = a + b + tmpNode.val;
                if (temp < 10) {
                    // 如果和小于10，则把和赋给当前节点
                    tmpNode.val = temp;
                    // 如果l1和l2都为空，则不再产生下一个节点
                    if (l1 != null || l2 != null) {
                        tmpNode.next = new ListNode(0);
                    }
                } else {
                    // 如果和大于10，则把和的个位赋给当前节点
                    tmpNode.val = temp - 10;
                    // 产生下一个值为1的节点
                    tmpNode.next = new ListNode(1);
                }
                // 指向下一个节点
                tmpNode = tmpNode.next;
            }
        }
        return result;
    }

    private void printListNode(ListNode listNode) {
        StringBuilder sb = new StringBuilder();
        while (listNode != null) {
            sb.append(listNode.val);
            listNode = listNode.next;
        }
        System.out.println(sb.toString());
    }

    /**
     * 单向链表
     */
    class ListNode {
        int val;
        ListNode next;

        ListNode(int x) {
            val = x;
        }
    }
}
